import streamlit as st
from PIL import Image

st.set_page_config(page_title="小工具中心", page_icon="🚀", layout="wide")

st.title("🚀 小工具中心")
st.caption("作者：何双新 ｜ 技术栈：Python + Streamlit")

st.markdown("## 🔥 精选小工具")

# 小工具列表（可以自己增加）
tools = [
    {
        "name": "Tensor 可视化器",
        "description": "可视化不同维度的PyTorch张量",
        "page_link": "1_Tensor可视化",
        "emoji": "📊"
    },
    {
        "name": "图片转Base64",
        "description": "上传图片，一键生成Base64编码",
        "page_link": "2_图片转Base64",
        "emoji": "🖼️"
    },
    {
        "name": "文本摘要生成器",
        "description": "使用AI提炼文章精华摘要",
        "page_link": "3_文本摘要生成",
        "emoji": "📝"
    }
]

# 三列布局
cols = st.columns(3)

for idx, tool in enumerate(tools):
    with cols[idx % 3]:
        st.subheader(f"{tool['emoji']} {tool['name']}")
        st.markdown(f"**{tool['description']}**")
        st.markdown(f"[👉 进入使用](./{tool['page_link']})")
        st.divider()
